#零样本提示（Zero-Shot）
# pip install openai
#提示词在大模型应用中直接使用，也可以看到效果
from openai import OpenAI
import os
from dotenv import load_dotenv  # pip install python-dotenv -i https://pypi.tuna.tsinghua.edu.cn/simple
load_dotenv()
client = OpenAI()

prompt = """
将文本分类为中性、负面或正面。
文本：我认为这家餐馆的菜品一般。
情感：
"""
# 在上面的提示中，我们没有向模型提供任何示例——这就是零样本能力的作用。
def get_completion(prompt, model="qwen-max-2025-01-25"):
    messages = [{"role": "user", "content": prompt}]
    response = client.chat.completions.create(
        model=model,
        messages=messages,
        temperature=0,  # 模型输出的随机性，0 表示随机性最小
    )
    return response.choices[0].message.content
print(get_completion(prompt))